Easiest way to split JSON file using Python